import type { Settings as LayoutSettings } from '@ant-design/pro-layout';
import { PageLoading } from '@ant-design/pro-layout';
// @ts-ignore
import type { RunTimeLayoutConfig } from 'umi';
import { history } from 'umi';
import type { Context } from 'umi-request';
import RightContent from '@/components/RightContent';
import Footer from '@/components/Footer';
import { currentUser as queryCurrentUser } from './services/ant-design-pro/api';
import defaultSettings from '../config/defaultSettings';
// @ts-ignore
import type { RequestConfig } from '@@/plugin-request/request';
import { WebsokcetOps } from '@/services/ws/websokcetops';
import { message } from 'antd';

const isDev = process.env.NODE_ENV === 'development';
const loginPath = '/user/login';

/** 获取用户信息比较慢的时候会展示一个 loading */
export const initialStateConfig = {
  loading: <PageLoading />,
};

/**
 * @see  https://umijs.org/zh-CN/plugins/plugin-initial-state
 * */
export async function getInitialState(): Promise<{
  settings?: Partial<LayoutSettings>;
  currentUser?: API.CurrentUser;
  loading?: boolean;
  fetchUserInfo?: () => Promise<API.CurrentUser | undefined>;
}> {
  const fetchUserInfo = async () => {
    try {
      const msg = await queryCurrentUser();
      // @ts-ignore
      if (process.env.NODE_ENV == 'development') {
        new WebsokcetOps().connection('ws://127.0.0.1:8585/ws').setAutoReconnection();
      } else {
        const noSsl = window.location.href.split(':')[0] == 'http';
        const httpScheme = noSsl ? 'http://' : 'https://';
        const domain =
          window.location.href.replace(httpScheme, '').split('/')[0].replace('/', '') + '/ws';
        new WebsokcetOps()
          .connection((noSsl ? 'ws' : 'wss') + '://' + domain)
          .setAutoReconnection();
      }
      return msg;
    } catch (error) {
      history.push(loginPath);
    }
    return undefined;
  };
  // 如果不是登录页面，执行
  if (history.location.pathname !== loginPath) {
    const currentUser = await fetchUserInfo();
    return {
      fetchUserInfo,
      currentUser,
      settings: defaultSettings,
    };
  }
  return {
    fetchUserInfo,
    settings: defaultSettings,
  };
}

// ProLayout 支持的api https://procomponents.ant.design/components/layout
// @ts-ignore
export const layout: RunTimeLayoutConfig = ({ initialState, setInitialState }) => {
  return {
    rightContentRender: () => <RightContent />,
    disableContentMargin: false,
    waterMarkProps: {
      content: initialState?.currentUser?.name,
    },
    footerRender: () => <Footer />,
    onPageChange: () => {
      const { location } = history;
      // 如果没有登录，重定向到 login
      if (!initialState?.currentUser && location.pathname !== loginPath) {
        history.push(loginPath);
      }
    },
    links: isDev
      ? [
          // <Link key="openapi" to="/umi/plugin/openapi" target="_blank">
          //   <LinkOutlined />
          //   <span>OpenAPI 文档</span>
          // </Link>,
          // <Link to="/~docs" key="docs">
          //   <BookOutlined />
          //   <span>业务组件文档</span>
          // </Link>,
        ]
      : [],
    menuHeaderRender: undefined,
    // 自定义 403 页面
    // unAccessible: <div>unAccessible</div>,
    // 增加一个 loading 的状态
    childrenRender: (children: any, props: any) => {
      // if (initialState?.loading) return <PageLoading />;
      return (
        <>
          {children}
          {/*{!props.location?.pathname?.includes('/login') && (*/}
          {/*  <SettingDrawer*/}
          {/*    disableUrlParams*/}
          {/*    enableDarkTheme*/}
          {/*    settings={initialState?.settings}*/}
          {/*    onSettingChange={(settings) => {*/}
          {/*      setInitialState((preInitialState: any) => ({*/}
          {/*        ...preInitialState,*/}
          {/*        settings,*/}
          {/*      }));*/}
          {/*    }}*/}
          {/*  />*/}
          {/*)}*/}
        </>
      );
    },
    ...initialState?.settings,
  };
};

const responseMiddleware = async (ctx: Context, next: () => void) => {
  // ctx.req.options.headers['Authorization'] =
  try {
    const userToken = localStorage.getItem('user_token');
    if (userToken && JSON.parse(userToken)) {
      const userTokenData = JSON.parse(userToken);
      // @ts-ignore
      ctx.req.options.headers['Authorization'] = userTokenData['access_token'];
    }
  } catch (e) {}

  // @ts-ignore

  await next();
  if (ctx.res.hasOwnProperty('data')) {
    if (ctx.res.data == undefined) {
      ctx.res.data = {};
    }
  }
  if (ctx.res.hasOwnProperty('errno')) {
    if (ctx.res.errno != 0) {
      if (ctx.res.errno == 10003) {
        history.push(loginPath);
        message.error('请登录');
      }
      throw new Error(ctx.res.msg);
    } else {
      ctx.res = ctx.res.data;
    }
  }
};

export const request: RequestConfig = {
  middlewares: [responseMiddleware],
};
